package q648_replaceWords;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class Solution_1 {

    /*
    简单的方式：
    先用一个hashSet存储所有的dict中的单词
    然后对于每一个sentence中的单词
    分别比对其所有前缀 是否在set中存在
    如果存在 则直接替换即可

    但此题为标准的字典树
    应该使用解法2
     */
    public String replaceWords(List<String> dictionary, String sentence) {
        Set<String> dictionarySet = new HashSet<String>(dictionary);


        String[] words = sentence.split(" ");
        for (int i = 0; i < words.length; i++) {
            String word = words[i];
            for (int j = 0; j < word.length(); j++) {
                if (dictionarySet.contains(word.substring(0, 1 + j))) {
                    words[i] = word.substring(0, 1 + j);
                    break;
                }
            }
        }

        return String.join(" ", words);

    }
}
